Added heat map functionality