changes to main, fixed what menu is displayed