增改函数，修改演示代码