Updates written in train and base version...