Fixed pylint complaints