some flake8 fixes
added figures for the README