More doc + ref in demo notebook