Code grooming (PEP8 cleanup)
Removed unused imports
Removed unused local variables