modify demo script