Edit according to PEP8