add connection prediction