List<WebElement> elements = driver.findElements(elementsLocator);
                classOfElements.clear();

                for (WebElement element : elements) {
                    String[] attributes = element.getAttribute("class").split("\\s+");

                    for (String attribute : attributes) {
                        classOfElements.add(attribute);

/*
    по идее - что было бы полезно увидеть в  classOfElements
    для каждого элемента = значение element.getAttribute("class")

    а в твоей реализации = будет не так
    если у одного из элементов будет несколько классов
    и тогда - по выведенному списку - будет тяжело понять - что за элементы и какие у каждого из них - классы

    так что - стоит classOfElements.add(...); перенести на уровень for (WebElement element : elements)
    и заполнять значением element.getAttribute("class")

    про название переменных  attribute и attributes
    тоже не лучшая идея)
    на самом деле - это cssClass & cssClasses
    если уточнить - то actualCssClass & actualCssClasses

    это надо поправить)
*/