more tweaks to model