Fix pylint and pep8 violation