增补注释，优化代码