优化代码，增加一些注释