Made it pylint compliant