modify code generator