add training and test data