now with heat map