update demo to match changes