代码整理

整理代码，添加注释