添加注释和完善代码结构