优化代码，增补注释