Changes to the TestBench - NOT WORKING