Update corr_pred.cc