Fix PEP8 violations in the python code