Add functionality for categories.