Add game state model code