Simplified code for returning to menu