improved compare_dists_above_below.py