--- a/add_sample_name_as_first_line.py	Sun Oct 21 09:37:06 2018 -0400
+++ b/add_sample_name_as_first_line.py	Fri Oct 26 17:52:33 2018 -0400
@@ -6,7 +6,7 @@
   the_parser = argparse.ArgumentParser(description="add label to first line of file")
   the_parser.add_argument('--input', required=True, action="store", type=str, help="input file")
   the_parser.add_argument('--output', required=True,  action="store", type=str, help="output file path")
-  the_parser.add_argument('--label', required=True, action="store", type=str, help="label to add in the first line")
+  the_parser.add_argument('--sample', required=True, action="store", type=str, help="label to add in the first line")
   args = the_parser.parse_args()
   return args
 
@@ -17,7 +17,7 @@
 #print >> output, args.label
 #print >> output, input
 
-sample_name = re.sub('(_1.fastq.gz|_2.fastq.gz|.fastq.gz)', '', args.label.rstrip().lstrip())
+sample_name = re.sub('(_1.fastq(.gz)*|_2.fastq(.gz)*|.fastq(.gz)*)', '', args.label.rstrip().lstrip())
 
 with open(args.input) as input:
 	with open(args.output, 'w') as output: