Make script PEP-8 compliant on imports and whitespace
Remove unused variables