minor changes to model fitting