Fix import of diff() in notebook_diff.py