Removed unused imports + PEP8