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

Fix imports for Python 3