MINOR : PEP8 + import

The title is explicit. :)