Add preliminary prediction code