Add DataGenerator