tidying up your code

微调一下代码