Cleanup codebook classes