Update DataGenerator.cs