Bugfix, typo
PEP8 fix
Removed unused import