Updated codegen