more updates to the demo