Checkpoint class half way done. Need to write the method for restore