Update generator
Add generator