pylint cleanup.
removed unused imports, whitespaces.