made changes to the unit_layer testbench generator