protection should actually be a list not a matrix.
Added a way to infect the graph.