Add GameState functionality