Tweaks & tidying