Add code generation