Some small changes to notebook_diff.py