fix issue #4. both py2 and py3 are supported now.
Merge pull request #5 from genzj/fix-issue-#4

fix issue #4. both py2 and py3 are supported now.