Adds modified DataGenerator