pylint cleanup