Fix predictor.fit diagram

Second demo notebook:

58

I moved this from another topic and removed the parts that I already fixed.

Should be fixed now.