added generator
Merge pull request #4 from csherratt/add_generator

added generator