Demo script updates