如何通过点击按钮来切换类？

<button id="toggleButton">切换类</button>
<div id="targetElement" class="class1">需要切换类的元素</div>

var toggleButton = document.getElementById("toggleButton");
var targetElement = document.getElementById("targetElement");

toggleButton.addEventListener("click", function() {
  targetElement.classList.toggle("class2");
});

.class1 {
  /* class1的样式 */
}

.class2 {
  /* class2的样式 */
}