Implement Forecast

Port the forecast part from Fortran to Python.