Updates to the code generator.