add DataGenerator and also split the data in Master.initialize