Update estimate.py