Updated demo to newest change in library