Fix setup.py
Fix flake8 warnings