Add a couple basic tests