Fix formatting to comply with PEP8