Adding a few more assert calls