Updates to codegen library