test data and training data done