- Refactored JIT code to have separate code generation functions