update code
Merge pull request #8 from fightingplane/master

update code