Add javascript so the demo on the page works