Add a couple of basic tests