Finetuned on gem/viggo dataset