improve the test demo