add training and test dataset