Added ability to add categories