Update code generator