[Task] split and rerun optimization nb for leaf and root dynamics.

[SarahAlidoost]

Before #62, update the notebooks. Also, since #59 is merged, we dont need to use config anymore.