Update generator.hs