[ENH] UFEM: Collsion kernel also works in 2D