Updated demo to match latest code release.