added TrainingData Class