update codegen.py