IMPORTANT: Change from row major order to column major order
Now this line should pass the testcase