Small PEP8 fixes