Adding Checkpoint usage to additional steps (already implemented for 01_index)"

[yuvalyaron]

As part of #186, checkpointing was added for the 01_index step only, now we would like to have it for other steps as well.

Checkpoints allow us to cache work that was already been done, so we don't repeat it, this can be useful for expensive calls, such as those to Open AI.