Finally satisfied with the toggle button