Adjusted demo file for new changes