Fix example and add comments