Switch class_name to source