Made code pylint compliant