Fixed the "Agent lost in space Syndrome"

I used to be passing an initialstategenerator and a game in the
visualization. Now the game is also the stategenerator..