Add toggle feature