Input: Category: animal 
Words: fly, train, cockroach, beetle, dive
Output: train