adjust the class names to match what is there