Updated testbench for ID