Added new category functionality