Puts imports back because silly @recon
Merge pull request #5 from kezz101/patch-1

Puts imports back because silly @recon