整理代码，添加注释