updated UI for edit screen