updated the categorization script