Added ability to add, remove categories