shorter code
Merge pull request #1 from axing21/master

shorter code