添加注释，代码优化