Categorize function works!