Update codegen