Reformated code to fix flake8 issues