Fixed pylint errors