Fix mypy errors, add notebook CI, clean up TODO

Fix mypy errors, add notebook CI, clean up TODO #450