modified prediction_model again