added training, test data