Got toggle button working