Update notebooks with new fitting framework