Minor PEP8 cleanup