fixed "done" button not returning to main menu