Add category functionality