Added an edit screen