Updated with category code
Merge pull request #8 from rickylaishram/v2

Updated with category code