Update associate_genes.py