add minor tweaks