Code formatting changes to satisfy flake8