Added StateBasedGame to keep track of screens