small change in the argpartition() call
changed col0 to k0 and col1 to k1 to be coherent with the argpartition() nomenclature, to indicate k0 is the first kth element and k1 the second