PEP8 cleaning