bugfix: add generator and output
Merge pull request #19 from gedare/bugfix

bugfix: add generator and output