Codebook update