updated testbench code