Update generator.cc