Updated Matrix Multiplication Code