Add GameState class