Add a DEMO link and update main code