Fixed pylint warnings