Do editing in viewWillDisappear