Added test prediction data