Refactor JIT basic-block identification