Update model_fit.R