add prediction to the documentation example