Comments update to get_num_of_infected_hosts in controller