Modify the dataGenerator