Small tweaks to game states