Update train_CB