Fixed some pylint warnings