Update toggle behavior