Add prediction parts