Как показать контейнер на клике с Localstorage?

Question

Я пытаюсь показать контейнер на клике с локальным хранилищем.Я имею в виду, что когда вы нажимаете кнопку, CSS classess: is-hidden и is-visible будут сохраняться и запоминаться браузером.

Это мой код:

var comments = document.getElementById("js-comments");
if (comments) {
    comments.addEventListener("click", function() {
        localStorage.setItem('comments', 'true');
        comments.classList.add("is-hidden");
        var container = document.getElementById("js-comments__inner");
        container.classList.add("is-visible");
        if (localStorage.getItem('comments') == 'true') {
            container.classList.add("is-visible");
        }
    });
}

HTML-разметка:

<div class="comments">
                    <button class="comments__button" id="js-comments">Load comments</button>
                    <div class="comments__inner" id="js-comments__inner">
                        <h3  class="h4">Comments</h3>
                    </div>
                </div>

Идея такова: когда вы нажимаете на элементы button и comments__inner, они получают CSS CSS.