make dataGen.py more general