bug fixed
Merge pull request #2 from hepochen/master

update