PEP8
Removed unused imports