Add DataGenRDD
more code