Correcting Nuisance Variation using Wasserstein Distance