small tidyup