updated demo file with latest code