added the prediction model