More main menu code