Fix return_menu() function