Made JIT implementation more compact