clean up imports
Merge pull request #33 from dannyroberts/fix-import

clean up imports