Added DataGenerator.py