Add ability to save current progress