Fix failing tests and warnings.
Re-run the example notebook
Update docs and version