优化部分代码，更新注释