add feature to define custom class names