Added save button to edit/view mode