modify code
Merge pull request #1 from htlxyz/master

modify code