Looped over trn and tst
Looped trn to 5