Work on prediction thing

Not done