updated viz function to color nodes by infected / not infected
updates to viz function and readme