Reformat code to match PEP guidelines