Minor notebook imports cleanups