Update demo script to v1.0.1