made 2D variant of KCSD work