add updated_prediction to bfgs