tidy up code

微整理代码