correct the compatibility between python3 and python2
Merge pull request #9 from sebdraven/master

Issue about compatibility