Added different states, including menu, load and game.