Updated the demo to work with list.