Added ability to add category