updating to match demo