Added menu and game over screens