Updated doc and checkpoints code