First attempt at adding assertions to testbench