Added unit test demo