Added segue logic for done button. Won't work without segue.