update to estimate.py