Added line for wait state in th testbench code.