Added training and test data