Add codegen support