on button click add remove class from div code example
Example: how to remove an class in javascript
document.getElementsByClassName("legend").style.display="none";
document.getElementsByClassName("legend").style.display="none";