Update demo to be consistent with public facing demo