After code generation