I fixed the add class bug