Added TestPrediction suite