Add heat map analysis