Fixing Pylint errors