присвоить элементу

Есть два типа объектов:

Немного погуглив, написал небольшую функцию, которая присваивает элементу новый класс, равный его data-атрибутуВ данном случае реализовал присвоение классу дочернему элементу, при клике на его родителя