addClass fix