Для управления атрибутом class удобнее пользоваться свойством classList . Оно содержит список всех классов элемента и обладает собственными методами, чтобы управлять этими классами.