JS показать, добавить забрать клас в зависимости от ширины экрана

Если нужно на JS показать <div> в зависимости от ширины экрана или на  JS добавить забрать клас в зависимости от ширины экрана

Пример - добавить стиль если экран меньше 800:

	<script>
		//функція визначає максимальну ширину в 600px і змінює класс - має бути в самому низу перед </body>
		const windowInnerWidth = window.innerWidth;
		function resizeListener() {
		if (windowInnerWidth < 600) {			
			var elem = document.querySelector('.body');
			// клас для екрану менше 600
			elem.className = 'body-mob';			
			} else {
			// клас для екрану більше 600
			var elem = document.querySelector('.body');
			elem.className = 'body-nomob';			
			}
		};
		resizeListener();
    </script>