Reformat code with PEP8