Describe the bug
The Visualization Tutorial now crashes on the last cell when run in Colab.
Likely something is going wrong by inputting the model for the histogram, because now it's a JupyterContainer object instead of a Mesa Model.

Expected behavior
Model runs without error and with correct histogram.
To Reproduce
Run the whole notebook in Colab.