Save edit on form submit too