Added TrainingData.scala