fixed the add class thing