Main_menu function changes