修改demo代码