[FIX] Code now respects PEP8 guidelines