Add data_generator.py