Addressing pylint warnings