Testbench work. Added a few control signals and changed the behavior of the testbench