Minor tweaks/fixes