add the test demo code