Add prediction assignment