This needs a complete clean-up. Lots of old stuff with deprecated code. Make new notebook using the latest interface, in order to showcase the model workflow for the toy model for the paper submission.