edit to original making paper_prediction