Add toggle functionality