Updated demo code and readme demo link.