add baseline prediction