modified heat map library