fixing up the menu in main cpp