Fixed imports and docstring snippet
Merge pull request #17 from micahhausler/master

Fixed imports and docstring snippet