Fix loop
PEP8