Fixes & tidying up