Some code added to MainMenu.cs