修改测试和演示代码