Add replay of train data