more tweaking to models