Adding a couple of basic tests