adding infected function