Merge pull request #16 from gsegura/master

fix issue while generating the category graph