clean-up

Cleaning up notebook so that it can be run for specific test cases.

#21373897041467