Updated notebook for tutorial