Update generator.js