Updating demo

updated demo to use the code from the last commit