some tweaks to the model