Fixed display_prediction_result