[IMP] Removed unused imports + PEP8 Compliant