minor demo modification