change the train data and the test data