Update Matrix Multiply solution for square assumption