Update forecast.py