Fixed error in CodeBook filename and made minor changes to the code