Mostly working. Saving iteration isn't working correctly.