Add MainMenu functionality