Added some basic tests based on the example