Added buffer update stage