Coloring by codebook column works, now just need to integrate labels