model update

tweaks to test dataset