fixed up the comments and changed the variable for the matrix to i

I guess this makes it look more like I've not just copied and pasted
from the example. And hey it's cool to play around with github etc