added tests for prediction