Fix some PEP8 warnings