started switching to TrainTestSplit