Add ability to save game state