Updating demo JS to match updated master