Add generator capability (github issue #1)