Update the class names