Added Checkpoint and cleaned up comments.