display_class updated with new structure