remove all classes then add the right one