set correct class names