Refactor JIT compiler