Made code of JIT implementation more regular