Corrects function display_menu