Fixed imports.
Merge pull request #9 from degenhard/master

Fixed imports.