[ ] make the setup common also with the Julia code by using a dictionary of settings and writing it to a JSON file (as in the Trixi example in PyMPDATA)
[ ] switch settings and the plot construction to match Shima's box-model plot
[ ] add asserts wrt analytical solution
[ ] try if all works on Colab, mybinder and ARM JupyterHub