Added category code

First pull request just for testing