code_gen updates