improve: 增加注释和美化代码