Minor tweaks/bug fixes