Fix bugs with find_infectious, find_susceptible