Fix formatting to conform to PEP8