Add feature to define custom class names