add class_name parameter