Switching Infection.m to Random Version