Update to include correct code book