added focus prediction