Improve Shift model.