Updated demo code
Updated link in docs