Add a checkpoint save before exiting