updating demo to match changes made