Add generator support.
Merge pull request #1 from rarkenin/patch-1

Add generator support.