Fixed the issues while making the code PEP8 compiant