Added categorization feature, but need to update views