Added tests for Predict