fixed code
Merge pull request #14 from CalvertYang/master

fixed code