Update noise_estimate.py