Final tweaks/fixes