Add support for Python 3.9+

The pre-commit hook fails for python 3.9 and above as pointed out in #1. Updating to nbformat version 5 fixes the issue for python 3.10 (and it should also work for python 3.9)