problem with csv file lengths for training set