Updated code to reflect latest changes in demo