Add functionality for categories