Quick PEP8 cleanup