Add errorbars, colors
Add fit_sota_model notebook