Merge pull request #4 from htlxyz/master

modify code