trying to make updated_prediction work