datagen.py