How to add or remove class on resize window

<script>
$(window).load(function() {
   
    var viewportWidth = $(window).width();
    if (viewportWidth < 480) {
            $(".cover-section").addClass("gallery-mobile");
    }
   
    $(window).resize(function () {
   
        if (viewportWidth < 480) {
            $(".cover-section").removeClass("gallery-mobile");
        }
    });
});
</script>