根据用户反馈，修改代码