refactor loadTrain and loadTrainTest