Added ability to save game