unit test just to make sure. Also changed dimensions of matrix in unit test not
to be square.