Clean up infect method
Convert m_infection from ArrayList into a HashMap in User class