added heat map script