Pylint cleanup
Pylint cleanup