updates to code gen