Update toggle() function