Update __init__.py

Make it work for Python 3.x
Merge pull request #3 from Ronggui/patch-1

Update __init__.py