modify code
Merge pull request #2 from htlxyz/master

modify code