Minor updates to kalman filter