添加注释，修改逻辑