Finish UI for edit