Adding toggle UI to the button