整理代码，主要添加了一些注释，删除了TODO