Added docblock, added tests for category