Minor updates to Kalman filter and the forecasting step