整理工程，优化代码