load test and train data