Fix flake8 warnings. Add tox with flake