Update to match pep8