Class (int column) { if (NotificationsTreeTable.DISPLAY_TYPE_COLUMN == column) { return NotificationDisplayType.class; } if (NotificationsTreeTable.LOG_COLUMN == column) { return Boolean.class; } if (NotificationsTreeTable.READ_ALOUD_COLUMN == column) { return Boolean.class; } return TreeTableModel.class; }