added TrainingSet class