Implement blocks in JIT