整理代码。