Fix formatting to adhere to PEP8