updated codebook to look nicer