fit_yreg function

Fit a model for the outcome given the treatment, mediator, and covariates.