Added a unit test of the example provided in the readme file