flake8 style clean up