Fix example and docs