diff --git a/functions.R b/functions.R
index 2e5357ed9458b640a47255d3fc567a45bb73392d..4e60496fbe007498c23fce3969fdfe1237f2923c 100644
--- a/functions.R
+++ b/functions.R
@@ -237,7 +237,7 @@ umapAllKit <- function(umapAll, merged_cell_annotation, annot, colorScale = FALS
   return(plot)
 }
 
-umapExpr <- function(umapAll, gene, correctedAll, annot, dataset="Dataset", sampleAnnot, main, violin_plot = FALSE, pointSize = 3, colorScale = c("white","yellow","red","purple")){
+umapExpr <- function(umapAll, gene, correctedAll, annot, dataset="Dataset", sampleAnnot, main, violin_plot = FALSE, pointSize = 3, ...){
   #' Plot UMAP with numeric variable 
   #' 
   #' @description WRAPPER
@@ -390,7 +390,7 @@ getColors <- function() {
   return(colors_list)
 }
 
-plotCompressedUmapExpr <- function(dir_path, pdf_path, names_list, correctedAll, umapAll, merged_cell_annotation, annot = "res.0.75", dataset = "Dataset", pointSize=2, nrow = 3, ncol = 3, colorScale=c("white","yellow","red","purple")) {
+plotCompressedUmapExpr <- function(dir_path, pdf_path, names_list, correctedAll, umapAll, merged_cell_annotation, annot = "res.0.75", dataset = "Dataset", pointSize=2, nrow = 3, ncol = 3, ...) {
   plot_list_compressed <- list()
   plot_list <- list()
   for (marker in names_list){