Added an edit mode