added limited infection functionality