1) fix the print out problem for all the train data