Fix script structure for count_transcripts_for_genes.