Add test for columnwise
Correct the expression