Added a simple test to check the example is working