modified estimate_K.py