Create DataGenerator.py