some tweaks, adding more models