unused imports removed
PEP8 fixes :smiley: