Add some basic tests