Added training data generator.