add edit_button to viewer