added the loop and a little cheat, after executing training data set program will shuffle the set and train on shuffled version and it does it 5 times, apparently it helps a lot :)