Update DataGenerator.h