FIx
SavePrediction
Updated doc