Basic implementation of total_infection
Implemented limited infection