Test bench creation

Making testBench have code inside!