Create forecast.py