Small change to Codebook