add the train data set