小修改一下师弟提交的代码，有意见欢迎讨论。