Add interpolation='nearest' to heatmap ploting function