Fix imports for Python 3
Merge pull request #2 from ivan/master

Fix imports for Python 3