Put my code in for the matrix function instead of other stuff

It was the assignment and therefore good.