整理代码,补点注释