Code book update