added a button for editing
edit now works as expected