Update generate.go