some small updates to demo