beautify, add heatmap?