【优化】整理代码