Added game states, button to start and restart game