Fix import for Python 3
Merge pull request #11 from sherbang/patch-1

Fix import for Python 3