Adding code generation