Reformat code to be PEP 8 compliant