More comments for checkpointing, fixed small bug #88

Unit tests pass