Added controller for in game state