Fixed codeBook again