improve code style
Merge pull request #2 from zzz6519003/patch-1

improve code style