PEP8 fixes