Added category feature
Merge pull request #8 from benlikescode/patch-1

Added category feature