Add simple unit test for example