Updated demo to use new waddell API