Fix  Issue #3
Merge pull request #4 from samtstern/fix-imports

Fix Issue #3