updated categorization script