Modified code and added GameState functionality