Added function to extract visitation prob.