Adding a couple of more assertions