Use rho as parameter
Fix test cases
Test nonzero cols