Remove unused module
PEP8 fix