added functionality for mask could be called with original umap