Update README and fix flake8