Made code compliant with PEP8