[Update] 整理代码