pylint warnings fix