Merge pull request #17 from franklixuefei/master

Merged Chao's commit
fixed minor bug