Improve code generation.