added in predictions on test data