Updated the saveToDisk method.