Added the unwind action for the "done" button