Obtaining training and test data sets

@@ -68,12 +68,13 @@ if __name__ == "__main__":
                     for line in iFile:
                         line = line.strip('\r\n')
                         trainingDataset.append(line)
-            if counter > 70 and counter <= 100:
+            elif counter > 70 and counter <= 100:
                 print("   Joining file {} to test data set".format(file))
                 with open(os.path.join(path, file), "r", encoding="utf-8", errors="replace") as iFile:
                     for line in iFile:
                         line = line.strip('\r\n')
                         testDataset.append(line)
+            counter += 1
     with open(os.path.join(options.outputPath, options.trainingFile), "w", encoding="utf-8", errors="replace") as oFile:
         for line in trainingDataset:
             oFile.write("{}\n".format(line))