small tidying up