added heatmap, stats, loading utils