Added main menu and game over screen