Fixed formating to make flake8 happy