Fixed bug and removed redundant imports in heatmap