updated layout for edit screen