flake8 fixes, add TODOs