Add a checkpoint creation class