Code gen changes