Update docs for predict()