Updated docs and demo