created main file
added total_infection