Update unit tests notebook