Fix setup and adjust for flake8