New loop in print_menu and moved exit inside print_menu