Update DataGenerator