Update generator
Merge pull request #50 from branflake2267/brandon_generator

Update generator