Demo script fix
Updated demo script