Updated Demo file to reflect API changes