Changed return type of display_menu() to int