make_prediction still glitchy