Added the main functionality of total infection