[demo/docs] Update notebook docs & minor demo update / fix (#1755)

[demo/docs] Update notebook docs & minor demo update / fix (#1755) #71