change code
Merge pull request #4 from htoooth/master

change code