(Somewhat) fixed the main menu