Updated demo script