corrected imports
Merge pull request #2 from Drezil/master

corrected imports