Update demo for new API

[ci skip]
Wording

[ci skip]