Cloned and adapted from the simplified version