* Removed unused imports
* PEP8 compliance