Create DataGenerator

I cant push yet so I copied and pasted my code here.