fix example in initial code