updated demo file with tests