Minor PEP8 tweaks