Formatted and styled code to pass flake8