Improve notebooks2sphinx.py