WIP fix PEP8