Added test for matrix mul function