updated demo to work better