Add missing imports
Merge pull request #5 from JeroenDeDauw/missingimp

Add missing imports