My submission for the matrix assignment