Little PEP8 fixes