Better mangement of the toggle button for page builder