Everything almost works. Saving checkpoint.