整理代码，增加注释