Add more comments.

Sketch out how to add checkpointing.