Update class names to match site