Implement unwind segues for correct returning to the main controller