Added generic to testbench