updated demo
minor changes pre code