Little PEP8 cleanup