Updated link to code in demo