Switch to className instead of classList.add for newly generated elements