Fixing some pylint warnings