Small edit to samples notebook