Added ability to get categories