Change imports
Merge pull request #4 from TheCricket/patch-1

Change imports