Update codegen.hs