Minor cleanup in find_critical_bandwidth.py