changed the way the prediction works

still not working with prediction