Modified the demo to include output