make code better

make code better
Merge pull request #59 from wenjun1055/master

make code better