Add reading utils, start checkpointing