Added ability to edit category