Update notebooks

The tutorial was done on 0.13, new versions introduced a few changes to notebooks, and the upcoming 0.14 will add changes too. We need to update this.