-- Using only required number of colors in heatmap
-- Removed unnecessary logs