minor update to the demo