Updated demo according to data attribute changes