Fix the code formatting to match pep8