updated w/ yhat