Restored datagen.py, to generate random data