Add option to create a 3D UMAP
Add option to create a 3D UMAP