update the TODO
update the demo