Built Population class and added Infected