Added GameState class
Changed Main class