修改代码，添加一些逻辑