Create compute_highest_affinity.py