Converted the code to PEP8 compliant