update demo to match repo