modify code

优化