Added imports for clarity
Merge pull request #9 from WoLpH/patch-1

Added imports for clarity