Updating the code to format pep8.