Changed from arrayList<Train> to arrayList<String>