Updated generator.go.