update demo with code