added 1d and 2d wasserstein functions