Converted the code to PEP8 compiant