Update generator.coffee