Added print_menu function