Added assert testing