Added return statement to main menu to fix loop bug