Tidy up categorization