Update demo script

Update to make demo project runnable .