adding categorization implementation