Fix docs
demo changes