ENH: Add datagenerator.py