Add toggle switch instead of two separate buttons