little change to python part
Merge pull request #21 from GuoSongtao/master

little change for Python wrapper