Add DUT and testbench code