Added generator.go