Add save and cancel button for edit mode