updated demo to show added functionality