Pylint cleanup!