Update PickleStepDataLoader.cls