Address pylint warnings