how to add remove div in javascript code example

Example: remove div javascript

var div = document.getElementById('div-01');
div.remove();