Fixes & tweaks