heatmap traces working correctly