Update compute_coherence.py