Small PEP8 and style fixes

* Fixed a typo
* Fixed some PEP8 issue
* Removed some stale imports