Adding id as class for fine-tuning