adding code to demo