Updated generated code