couple of minor tweaks