Update edit_screen.xml