Allow configuration of class_name